/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright (c) 1997-2018 Oracle and/or its affiliates. All rights reserved.
 *
 * The contents of this file are subject to the terms of either the GNU
 * General Public License Version 2 only ("GPL") or the Common Development
 * and Distribution License("CDDL") (collectively, the "License").  You
 * may not use this file except in compliance with the License.  You can
 * obtain a copy of the License at
 * https://oss.oracle.com/licenses/CDDL+GPL-1.1
 * or LICENSE.txt.  See the License for the specific
 * language governing permissions and limitations under the License.
 *
 * When distributing the software, include this License Header Notice in each
 * file and include the License file at LICENSE.txt.
 *
 * GPL Classpath Exception:
 * Oracle designates this particular file as subject to the "Classpath"
 * exception as provided by Oracle in the GPL Version 2 section of the License
 * file that accompanied this code.
 *
 * Modifications:
 * If applicable, add the following below the License Header, with the fields
 * enclosed by brackets [] replaced by your own identifying information:
 * "Portions Copyright [year] [name of copyright owner]"
 *
 * Contributor(s):
 * If you wish your version of this file to be governed by only the CDDL or
 * only the GPL Version 2, indicate your decision by adding "[Contributor]
 * elects to include this software in this distribution under the [CDDL or GPL
 * Version 2] license."  If you don't indicate a single choice of license, a
 * recipient has the option to distribute your version of this file under
 * either the CDDL, the GPL Version 2 or to extend the choice of license to
 * its licensees as provided above.  However, if you add GPL Version 2 code
 * and therefore, elected the GPL Version 2 license, then the option applies
 * only if the new code is made subject to such option by the copyright
 * holder.
 *
 * 重写 JavaMail 包下，该类的原因是：
 * 错误的：filename==?utf-8?b?5rGH546H5oql5ZGKXzIwMTktMDctMTUueGxzeA==?=
 * 正确的：filename="=?utf-8?b?5rGH546H5oql5ZGKXzIwMTktMDctMTUueGxzeA==?="
 * filename 表示邮件中附件的名字。极少数情况下 filename 没有加 双引号 导致程序异常。一下if语句用来处理这种异常信息。
 */

package javax.mail.internet;

import com.song.util.CommonConstant;
import com.sun.mail.util.PropUtil;

/**
 * This class represents a MIME ContentDisposition value. It provides
 * methods to parse a ContentDisposition string into individual components
 * and to generate a MIME style ContentDisposition string.
 *
 * @author  John Mani
 */

public class ContentDisposition {

    private static final boolean contentDispositionStrict =
            PropUtil.getBooleanSystemProperty("mail.mime.contentdisposition.strict", true);

    private String disposition; // disposition
    private ParameterList list;	// parameter list

    /**
     * No-arg Constructor.
     */
    public ContentDisposition() { }

    /**
     * Constructor.
     *
     * @param	disposition	disposition
     * @param	list	ParameterList
     * @since		JavaMail 1.2
     */
    public ContentDisposition(String disposition, ParameterList list) {
        this.disposition = disposition;
        this.list = list;
    }

    /**
     * Constructor that takes a ContentDisposition string. The String
     * is parsed into its constituents: dispostion and parameters.
     * A ParseException is thrown if the parse fails.
     *
     * @param	s	the ContentDisposition string.
     * @exception	ParseException if the parse fails.
     * @since		JavaMail 1.2
     */
    public ContentDisposition(String s) throws ParseException {
        HeaderTokenizer h = new HeaderTokenizer(s, HeaderTokenizer.MIME);
        HeaderTokenizer.Token tk;

        // First "disposition" ..
        tk = h.next();
        if (tk.getType() != HeaderTokenizer.Token.ATOM) {
            if (contentDispositionStrict) {
                throw new ParseException("Expected disposition, got " +
                        tk.getValue());
            }
        } else {
            disposition = tk.getValue();
        }

        // Then parameters ..
        String rem = h.getRemainder();
//	System.out.println("1修改前： " + rem);


//	错误的：filename==?utf-8?b?5rGH546H5oql5ZGKXzIwMTktMDctMTUueGxzeA==?=
//	正确的：filename="=?utf-8?b?5rGH546H5oql5ZGKXzIwMTktMDctMTUueGxzeA==?="
//	filename 表示邮件中附件的名字。极少数情况下 filename 没有加 双引号 导致程序异常。一下if语句用来处理这种异常信息。
        if(rem != null && rem.indexOf("\"") == -1) {

            String remUpperCase = rem.toUpperCase();

            int remIndexOf = remUpperCase.indexOf(CommonConstant.UTF_8_B);

            if(remIndexOf != -1) {
                String remSuffix = rem.substring(remIndexOf);//=?utf-8?b?5rGH546H5oql5ZGKXzIwMTktMDctMTUueGxzeA==?=
                String remPrefix = rem.substring(0, remIndexOf);// ; filename=
                remPrefix = remPrefix + "\"" + remSuffix;

                int remIndexOfEnd = remUpperCase.indexOf(CommonConstant.UTF_8_END);// 63
                String remSuffixEnd = rem.substring(remIndexOfEnd + CommonConstant.UTF_8_END.length());
                remSuffixEnd =  remSuffixEnd +"\"";

                rem = remPrefix + remSuffixEnd;
            } else {

                remIndexOf = remUpperCase.indexOf(CommonConstant.UTF_8_Q);
                if(remIndexOf != -1) {
                    String remSuffix = rem.substring(remIndexOf);
                    String remPrefix = rem.substring(0, remIndexOf);
                    remPrefix = remPrefix + "\"" + remSuffix;

                    int remIndexOfEnd = remUpperCase.indexOf(CommonConstant.UTF_8_END);// 63
                    String remSuffixEnd = rem.substring(remIndexOfEnd + CommonConstant.UTF_8_END.length());
                    remSuffixEnd =  remSuffixEnd +"\"";

                    rem = remPrefix + remSuffixEnd;
                }
            }
        }


        if (rem != null) {
            try {
                list = new ParameterList(rem);
            } catch (ParseException px) {
                if (contentDispositionStrict) {
                    throw px;
                }
            }
        }
    }

    /**
     * Return the disposition value.
     * @return the disposition
     * @since		JavaMail 1.2
     */
    public String getDisposition() {
        return disposition;
    }

    /**
     * Return the specified parameter value. Returns <code>null</code>
     * if this parameter is absent.
     *
     * @param	name	the parameter name
     * @return	parameter value
     * @since		JavaMail 1.2
     */
    public String getParameter(String name) {
        if (list == null)
            return null;

        return list.get(name);
    }

    /**
     * Return a ParameterList object that holds all the available
     * parameters. Returns null if no parameters are available.
     *
     * @return	ParameterList
     * @since		JavaMail 1.2
     */
    public ParameterList getParameterList() {
        return list;
    }

    /**
     * Set the disposition.  Replaces the existing disposition.
     * @param	disposition	the disposition
     * @since		JavaMail 1.2
     */
    public void setDisposition(String disposition) {
        this.disposition = disposition;
    }

    /**
     * Set the specified parameter. If this parameter already exists,
     * it is replaced by this new value.
     *
     * @param	name	parameter name
     * @param	value	parameter value
     * @since		JavaMail 1.2
     */
    public void setParameter(String name, String value) {
        if (list == null)
            list = new ParameterList();

        list.set(name, value);
    }

    /**
     * Set a new ParameterList.
     * @param	list	ParameterList
     * @since		JavaMail 1.2
     */
    public void setParameterList(ParameterList list) {
        this.list = list;
    }

    /**
     * Retrieve a RFC2045 style string representation of
     * this ContentDisposition. Returns an empty string if
     * the conversion failed.
     *
     * @return	RFC2045 style string
     * @since		JavaMail 1.2
     */
    @Override
    public String toString() {
        if (disposition == null)
            return "";

        if (list == null)
            return disposition;

        StringBuilder sb = new StringBuilder(disposition);

        // append the parameter list
        // use the length of the string buffer + the length of
        // the header name formatted as follows "Content-Disposition: "
        sb.append(list.toString(sb.length() + 21));
        return sb.toString();
    }
}
